查看原文
其他

太牛了!这个神奇工具让你的照片秒变宫崎骏漫画

小秋 开源前哨 2022-09-24

简介

AnimeGANv2 是开源项目 AnimeGAN 的改进版本,可以将图片和视频转换成漫画风格的工具。采用的是神经风格迁移 + 生成对抗网络(GAN)的组合,转换速度快。

AnimeGANv2 改进的内容主要包括以下几点:

  1. 解决生成图像中高频伪影的问题
  2. 易于训练,能达到对应论文中的效果
  3. 进一步减少生成器网络的参数数量(生成器大小:8.17Mb),精简版的生成器会更小
  4. 使用新的高质量风格数据,尽可能来自 BD 电影



项目地址是:

https://github.com/TachibanaYoshino/AnimeGANv2

用法

依赖

  • python 3.6
  • tensorflow-gpu 1.15.0
  • opencv
  • tqdm
  • numpy
  • glob
  • argparse
  • onnxruntime

用法

  1. 推理
python test.py --checkpoint_dir checkpoint/generator_Hayao_weight --test_dir dataset/test/HR_photo --save_dir Hayao/HR_photo
  1. 将视频转换为动画
python video2anime.py --video video/input/お花見.mp4 --checkpoint_dir checkpoint/generator_Hayao_weight --output video/output
  1. 训练
  • 下载 vgg19
https://github.com/TachibanaYoshino/AnimeGAN/releases/tag/vgg16%2F19.npy
  • 下载数据集
https://github.com/TachibanaYoshino/AnimeGAN/releases/tag/dataset-1
  • 边缘平滑
python edge_smooth.py --dataset Hayao --img_size 256
  • 训练
python train.py --dataset Hayao --epoch 101 --init_epoch 10
  • 提取生成器的权重
python get_generator_ckpt.py --checkpoint_dir ../checkpoint/AnimeGANv2_Shinkai_lsgan_300_300_1_2_10_1 --style_name Shinkai

结果

今敏动画风格

宫崎骏动漫风格

新海诚动漫风格


- EOF - 


更多优秀开源项目(点击下方图片可跳转)




开源前哨

日常分享热门、有趣和实用的开源项目。参与维护10万+star 的开源技术资源库,包括:Python, Java, C/C++, Go, JS, CSS, Node.js, PHP, .NET 等

关注后获取

回复 资源 获取 10万+ star 开源资源



分享点赞在看

支持我们分享更多优秀开源项目,谢谢!

您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存